import boto3
import os
import sys
import uuid
from urllib.parse import unquote_plus
from PIL import Image, ExifTags

s3_client = boto3.client('s3')

def resize_image(image_path, resized_path):
    max_file_size = 3 * 1024 * 1024  # 3MB
    reduction_step = 0.9  # 10%씩 줄임

    with Image.open(image_path) as image:
        # EXIF 데이터에서 회전 정보 읽기
        try:
            for orientation in ExifTags.TAGS.keys():
                if ExifTags.TAGS[orientation] == 'Orientation':
                    break
            exif = image._getexif()
            if exif is not None:
                orientation_value = exif.get(orientation)
                if orientation_value is not None:
                    if orientation_value == 3:
                        image = image.rotate(180, expand=True)
                    elif orientation_value == 6:
                        image = image.rotate(270, expand=True)
                    elif orientation_value == 8:
                        image = image.rotate(90, expand=True)
        except Exception as e:
            print(f"Failed to read EXIF data: {e}")

        # 원본 이미지 포맷 확인
        image_format = image.format
        if image_format is None:
            image_format = 'JPEG'  # 기본적으로 JPEG로 저장

        # 이미지 리사이즈: 10%씩 줄이면서 파일 크기를 확인
        while True:
            # 현재 상태의 이미지 임시로 저장
            image.save(resized_path, image_format)

            # 파일 크기 확인
            file_size = os.path.getsize(resized_path)

            # 파일 크기가 3MB 이하인지 확인
            if file_size <= max_file_size:
                print(f"Final image size: {file_size / 1024 / 1024:.2f} MB")
                break

            # 이미지 크기 90%로 줄이기
            new_size = tuple(int(x * reduction_step) for x in image.size)
            image = image.resize(new_size, Image.ANTIALIAS)

        # 최종 이미지 저장
        image.save(resized_path, image_format)

def lambda_handler(event, context):
    for record in event['Records']:
        bucket = record['s3']['bucket']['name']
        key = unquote_plus(record['s3']['object']['key'])
        tmpkey = key.replace('/', '')
        download_path = '/tmp/{}{}'.format(uuid.uuid4(), tmpkey)
        upload_path = '/tmp/resized-{}'.format(tmpkey)

        # S3에서 파일 다운로드
        s3_client.download_file(bucket, key, download_path)

        try:
            # 이미지 리사이즈
            resize_image(download_path, upload_path)

            # 원본 파일의 메타데이터 가져오기
            original_object = s3_client.head_object(Bucket=bucket, Key=key)
            original_content_type = original_object['ContentType']

            # 기존 경로 : origin/profile/,,,
            # 수정 경로 : profile/,,,
            upload_key = key.replace('origin/', '')
            print(f"Uploading resized image to: profile/{upload_key}")

            # 파일을 퍼블릭으로 업로드하고 Content-Type 및 Content-Disposition 설정
            s3_client.upload_file(
                upload_path,
                bucket,
                upload_key,
                ExtraArgs={
                    'ACL': 'public-read',
                    'ContentType': original_content_type,  # 기존 파일의 Content-Type 사용
                    'ContentDisposition': 'inline'  # 브라우저에서 바로 열리도록 설정
                }
            )

            # 기존 파일 삭제
            print(f"Deleting original file: {key}")
            s3_client.delete_object(Bucket=bucket, Key=key)

        except Exception as e:
            print(f"Error processing file {key}: {e}")
            raise e